import os
from openai import OpenAI

client = OpenAI(
    api_key="sk-bf223be5c39d4d4bbdbca784d92f54ff",
    base_url="example_base_url",
)


def answer(text):
    prompt = f"跟进{text}，推理并输出对应的化学式。例如：水为“H2O”。仅输出化学式，不需要分析过程或解释。识别不出的输出 H20"
    completion = client.chat.completions.create(
        model="qwen-plus",
        messages=[
            {'role': 'system',
             'content': '你是一位化学专家，精通各种化学原理、化学式、反应过程等。你能够迅速识别和分析化学问题，并提供准确的化学相关信息和解答。'},
            {'role': 'user', 'content': prompt}],
    )
    print(completion.model_dump_json())
    return completion.choices[0].message.content
